package models

import (
	"gorm.io/gorm"
)

type News struct {
	gorm.Model
	Title   string `gorm:"type:varchar(30);not null;comment:'新闻标题'"`
	Address string `gorm:"type:varchar(100);not null;comment:'所属区域'"`
	TypesId int    `gorm:"type:int;not null;comment:'新闻类型ID'"`
	Author  string `gorm:"type:varchar(30);not null;comment:'发布人'"`
	Image   string `gorm:"type:varchar(100);not null;comment:'新闻图片'"`
	Time    string `gorm:"type:datetime;not null;comment:'发布时间'"`
	Pl      string `gorm:"type:enum('人工审核','电脑审核');not null;comment:'评论设置'"`
	Content string `gorm:"type:varchar(100);not null;comment:'文章正文'"`
	Status  int    `gorm:"type:tinyint;not null;default:1;comment:'状态 1已发布 2未发布'"`
}

type Types struct {
	gorm.Model
	Name string `gorm:"type:varchar(30);not null;comment:'新闻类型'"`
}

// 类型添加
func (u *Types) CreateTypes() error {
	return DB.Create(&u).Error
}

// 类型查询
func (u *Types) GetNewsBy() error {
	return DB.Where("id = ?", u.ID).Limit(1).Find(&u).Error
}

// 新闻查询
func (u *News) GetNewsBy() error {
	return DB.Where("id = ?", u.ID).Limit(1).Find(&u).Error
}

// 新闻添加
func (u *News) Create() error {
	return DB.Create(&u).Error
}

// 新闻删除
func (u *News) Del() error {
	return DB.Delete(&u).Error
}

func (u *News) List(keyword string, page, pageSize int) (list []*News, err error) {
	err = DB.Where("title like ?", "%"+keyword+"%").Scopes(Paginate(page, pageSize)).Find(&list).Error
	return
}

func Paginate(page, pageSize int) func(db *gorm.DB) *gorm.DB {
	return func(db *gorm.DB) *gorm.DB {
		if page <= 0 {
			page = 1
		}

		switch {
		case pageSize > 100:
			pageSize = 100
		case pageSize <= 0:
			pageSize = 10
		}

		offset := (page - 1) * pageSize
		return db.Offset(offset).Limit(pageSize)
	}
}
